package br.ucb.sirad.managedBean;

import java.util.List;

import javax.faces.bean.ManagedBean;

import org.primefaces.model.DefaultTreeNode;
import org.primefaces.model.TreeNode;

import br.ucb.sirad.beans.Categoria;
import br.ucb.sirad.beans.Documento;
import br.ucb.sirad.dao.CategoriaDAO;
import br.ucb.sirad.dao.DocumentoDAO;

@ManagedBean
public class TreeBean {
	private TreeNode root;  
	  
    public TreeBean() {  

    	CategoriaDAO categoriaDao = new CategoriaDAO();
    	DocumentoDAO documentoDao = new DocumentoDAO();

    	List<Categoria> listaCategoria = categoriaDao.buscarArvore();
    	
    	root = new DefaultTreeNode("Root", null);  
    	
    	for(Categoria cat : listaCategoria)
    	{
    		TreeNode categoria = new DefaultTreeNode(cat.getNome(), root);
    		
    		documentoDao.buscaDocumentosCategoria(cat);
    		
    		for(Documento doc : documentoDao.buscaDocumentosCategoria(cat))
    		{
    			TreeNode documents = new DefaultTreeNode(doc.getNome(), categoria);  
    		}
            
    	}   
    	
    }  
  
    public TreeNode getRoot() {  
        return root;  
    }     
}
